'''使用numpy进行数组切分'''
import numpy as np
#创建一个数组
data = np.arange(10)
print("原始数据:",data)

#切分数组
train, test = np.split(data, [7])
print("训练集:",train)
print("测试集:",test)